package com.train.webadmin.app.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.hutool.core.collection.CollUtil;
import com.train.common.log.annotation.OperationLog;
import com.train.common.log.model.constant.SysOperationLogType;
import com.github.pagehelper.page.PageMethod;
import com.train.webadmin.app.vo.*;
import com.train.webadmin.app.dto.*;
import com.train.webadmin.app.model.*;
import com.train.webadmin.app.service.*;
import com.train.common.core.object.*;
import com.train.common.core.util.*;
import com.train.common.core.constant.*;
import com.train.common.core.annotation.MyRequestBody;
import com.train.webadmin.config.ApplicationConfig;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springdoc.api.annotations.ParameterObject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 试卷题目操作控制器类。
 *
 * @author xulei
 * @date 2025-03-07
 */
@Tag(name = "试卷题目管理接口")
@Slf4j
@RestController
@RequestMapping("/admin/app/tQuestionInfo")
public class TQuestionInfoController {

    @Autowired
    private TPaperInfoService tPaperInfoService;
    @Autowired
    private ApplicationConfig appConfig;
    @Autowired
    private TQuestionInfoService tQuestionInfoService;

    /**
     * 新增试卷题目数据。
     *
     * @param tQuestionInfoDto 新增对象。
     * @return 应答结果对象，包含新增对象主键Id。
     */
    @ApiOperationSupport(ignoreParameters = {"tQuestionInfoDto.id"})
    @SaCheckPermission("tQuestionInfo.add")
    @OperationLog(type = SysOperationLogType.ADD)
    @PostMapping("/add")
    public ResponseResult<Long> add(@MyRequestBody TQuestionInfoDto tQuestionInfoDto) {
        String errorMessage = MyCommonUtil.getModelValidationError(tQuestionInfoDto, false);
        if (errorMessage != null) {
            return ResponseResult.error(ErrorCodeEnum.DATA_VALIDATED_FAILED, errorMessage);
        }
        TQuestionInfo tQuestionInfo = MyModelUtil.copyTo(tQuestionInfoDto, TQuestionInfo.class);
        // 验证关联Id的数据合法性
        CallResult callResult = tQuestionInfoService.verifyRelatedData(tQuestionInfo, null);
        if (!callResult.isSuccess()) {
            return ResponseResult.errorFrom(callResult);
        }
        tQuestionInfo = tQuestionInfoService.saveNew(tQuestionInfo);
        return ResponseResult.success(tQuestionInfo.getId());
    }

    /**
     * 更新试卷题目数据。
     *
     * @param tQuestionInfoDto 更新对象。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @OperationLog(type = SysOperationLogType.UPDATE)
    @PostMapping("/update")
    public ResponseResult<Void> update(@MyRequestBody TQuestionInfoDto tQuestionInfoDto) {
        String errorMessage = MyCommonUtil.getModelValidationError(tQuestionInfoDto, true);
        if (errorMessage != null) {
            return ResponseResult.error(ErrorCodeEnum.DATA_VALIDATED_FAILED, errorMessage);
        }
        TQuestionInfo tQuestionInfo = MyModelUtil.copyTo(tQuestionInfoDto, TQuestionInfo.class);
        TQuestionInfo originalTQuestionInfo = tQuestionInfoService.getById(tQuestionInfo.getId());
        if (originalTQuestionInfo == null) {
            // NOTE: 修改下面方括号中的话述
            errorMessage = "数据验证失败，当前 [数据] 并不存在，请刷新后重试！";
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST, errorMessage);
        }
        // 验证关联Id的数据合法性
        CallResult callResult = tQuestionInfoService.verifyRelatedData(tQuestionInfo, originalTQuestionInfo);
        if (!callResult.isSuccess()) {
            return ResponseResult.errorFrom(callResult);
        }
        if (!tQuestionInfoService.update(tQuestionInfo, originalTQuestionInfo)) {
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST);
        }
        return ResponseResult.success();
    }

    /**
     * 删除试卷题目数据。
     *
     * @param id 删除对象主键Id。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.delete")
    @OperationLog(type = SysOperationLogType.DELETE)
    @PostMapping("/delete")
    public ResponseResult<Void> delete(@MyRequestBody Long id) {
        if (MyCommonUtil.existBlankArgument(id)) {
            return ResponseResult.error(ErrorCodeEnum.ARGUMENT_NULL_EXIST);
        }
        return this.doDelete(id);
    }

    /**
     * 批量删除试卷题目数据。
     *
     * @param idList 待删除对象的主键Id列表。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.delete")
    @OperationLog(type = SysOperationLogType.DELETE_BATCH)
    @PostMapping("/deleteBatch")
    public ResponseResult<Void> deleteBatch(@MyRequestBody List<Long> idList) {
        if (MyCommonUtil.existBlankArgument(idList)) {
            return ResponseResult.error(ErrorCodeEnum.ARGUMENT_NULL_EXIST);
        }
        for (Long id : idList) {
            ResponseResult<Void> responseResult = this.doDelete(id);
            if (!responseResult.isSuccess()) {
                return responseResult;
            }
        }
        return ResponseResult.success();
    }

    /**
     * 列出符合过滤条件的试卷题目列表。
     *
     * @param tQuestionInfoDtoFilter 过滤对象。
     * @param orderParam 排序参数。
     * @param pageParam 分页参数。
     * @return 应答结果对象，包含查询结果集。
     */
    @SaCheckPermission("tQuestionInfo.view")
    @PostMapping("/list")
    public ResponseResult<MyPageData<TQuestionInfoVo>> list(
            @MyRequestBody TQuestionInfoDto tQuestionInfoDtoFilter,
            @MyRequestBody MyOrderParam orderParam,
            @MyRequestBody MyPageParam pageParam) {
        if (pageParam != null) {
            PageMethod.startPage(pageParam.getPageNum(), pageParam.getPageSize(), pageParam.getCount());
        }
        TQuestionInfo tQuestionInfoFilter = MyModelUtil.copyTo(tQuestionInfoDtoFilter, TQuestionInfo.class);
        String orderBy = MyOrderParam.buildOrderBy(orderParam, TQuestionInfo.class);
        List<TQuestionInfo> tQuestionInfoList =
                tQuestionInfoService.getTQuestionInfoListWithRelation(tQuestionInfoFilter, orderBy);
        return ResponseResult.success(MyPageUtil.makeResponseData(tQuestionInfoList, TQuestionInfoVo.class));
    }

    /**
     * 导入主表数据列表。
     *
     * @param importFile 上传的文件，目前仅仅支持xlsx和xls两种格式。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.import")
    @OperationLog(type = SysOperationLogType.IMPORT)
    @PostMapping("/import")
    public ResponseResult<Void> importBatch(
            @RequestParam Boolean skipHeader,
            @RequestParam("importFile") MultipartFile importFile) throws IOException {
        String filename = ImportUtil.saveImportFile(appConfig.getUploadFileBaseDir(), null, importFile);
        // 这里可以指定需要忽略导入的字段集合。如创建时间、创建人、更新时间、更新人、主键Id和逻辑删除，
        // 以及一些存在缺省值且无需导入的字段。其中主键字段和逻辑删除字段不需要在这里设置，批量插入逻辑会自动处理的。
        Set<String> ignoreFieldSet = new HashSet<>();
        ignoreFieldSet.add("createTime");
        ignoreFieldSet.add("updateTime");
        ignoreFieldSet.add("createUserId");
        ignoreFieldSet.add("updateUserId");
        List<ImportUtil.ImportHeaderInfo> headerInfoList = ImportUtil.makeHeaderInfoList(TQuestionInfo.class, ignoreFieldSet);
        // 下面是导入时需要注意的地方，如果我们缺省生成的代码，与实际情况存在差异，请手动修改。
        // 1. 头信息数据字段，我们只是根据当前的主表实体对象生成了缺省数组，开发者可根据实际情况，对headerInfoList进行修改。
        ImportUtil.ImportHeaderInfo[] headerInfos = headerInfoList.toArray(new ImportUtil.ImportHeaderInfo[]{});
        // 2. 这里需要根据实际情况决定，导入文件中第一行是否为中文头信息，如果是可以跳过。这里我们默认为true。
        // 这里根据自己的实际需求，为doImport的最后一个参数，传递需要进行字典转换的字段集合。
        // 注意，集合中包含需要翻译的Java字段名，如: gradeId。
        Set<String> translatedDictFieldSet = new HashSet<>();
        translatedDictFieldSet.add("courseId");
        translatedDictFieldSet.add("paperId");
        List<TQuestionInfo> dataList =
                ImportUtil.doImport(headerInfos, skipHeader, filename, TQuestionInfo.class, translatedDictFieldSet);
        CallResult result = tQuestionInfoService.verifyImportList(dataList, translatedDictFieldSet);
        if (!result.isSuccess()) {
            // result中返回了具体的验证失败对象，如果需要返回更加详细的错误，可根据实际情况手动修改。
            return ResponseResult.errorFrom(result);
        }
        //TODO: 下面方法的第二个字段列表参数，用于判断导入的数据在数据表是否已经存在，存在则更新，否则插入，如全部新数据，则无需任何修改。
        tQuestionInfoService.saveNewOrUpdateBatch(dataList, CollUtil.newArrayList(), -1);
        return ResponseResult.success();
    }

    /**
     * 导出符合过滤条件的试卷题目列表。
     *
     * @param tQuestionInfoDtoFilter 过滤对象。
     * @param orderParam 排序参数。
     * @throws IOException 文件读写失败。
     */
    @SaCheckPermission("tQuestionInfo.export")
    @OperationLog(type = SysOperationLogType.EXPORT, saveResponse = false)
    @PostMapping("/export")
    public void export(
            @MyRequestBody TQuestionInfoDto tQuestionInfoDtoFilter,
            @MyRequestBody MyOrderParam orderParam) throws IOException {
        TQuestionInfo tQuestionInfoFilter = MyModelUtil.copyTo(tQuestionInfoDtoFilter, TQuestionInfo.class);
        String orderBy = MyOrderParam.buildOrderBy(orderParam, TQuestionInfo.class);
        List<TQuestionInfo> resultList =
                tQuestionInfoService.getTQuestionInfoListWithRelation(tQuestionInfoFilter, orderBy);
        // 导出文件的标题数组
        // NOTE: 下面的代码中仅仅导出了主表数据，主表聚合计算数据和主表关联字典的数据。
        // 一对一从表数据的导出，可根据需要自行添加。如：headerMap.put("slaveFieldName.xxxField", "标题名称")
        Map<String, String> headerMap = new LinkedHashMap<>(15);
        headerMap.put("id", "题目id");
        headerMap.put("courseIdDictMap.name", "课程id");
        headerMap.put("paperIdDictMap.name", "试卷id");
        headerMap.put("title", "题干");
        headerMap.put("types", "类型");
        headerMap.put("remark", "解析");
        headerMap.put("score", "分数");
        headerMap.put("degree", "难度");
        headerMap.put("createTime", "创建时间");
        headerMap.put("updateTime", "更新时间");
        headerMap.put("createUserId", "创建人id");
        headerMap.put("updateUserId", "修改人id");
        headerMap.put("deletedFlag", "删除标志");
        headerMap.put("correctAnswer", "正确答案");
        headerMap.put("options", "选项");
        ExportUtil.doExport(resultList, headerMap, "tQuestionInfo.xlsx");
    }

    /**
     * 查看指定试卷题目对象详情。
     *
     * @param id 指定对象主键Id。
     * @return 应答结果对象，包含对象详情。
     */
    @SaCheckPermission("tQuestionInfo.view")
    @GetMapping("/view")
    public ResponseResult<TQuestionInfoVo> view(@RequestParam Long id) {
        TQuestionInfo tQuestionInfo = tQuestionInfoService.getByIdWithRelation(id, MyRelationParam.full());
        if (tQuestionInfo == null) {
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST);
        }
        TQuestionInfoVo tQuestionInfoVo = MyModelUtil.copyTo(tQuestionInfo, TQuestionInfoVo.class);
        return ResponseResult.success(tQuestionInfoVo);
    }

    /**
     * 列出不与指定试卷题目存在多对多关系的 [考试试卷] 列表数据。通常用于查看添加新 [考试试卷] 对象的候选列表。
     *
     * @param id 主表关联字段。
     * @param tPaperInfoDtoFilter [考试试卷] 过滤对象。
     * @param orderParam 排序参数。
     * @param pageParam 分页参数。
     * @return 应答结果对象，返回符合条件的数据列表。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @PostMapping("/listNotInTPaperQuest")
    public ResponseResult<MyPageData<TPaperInfoVo>> listNotInTPaperQuest(
            @MyRequestBody Long id,
            @MyRequestBody TPaperInfoDto tPaperInfoDtoFilter,
            @MyRequestBody MyOrderParam orderParam,
            @MyRequestBody MyPageParam pageParam) {
        if (MyCommonUtil.isNotBlankOrNull(id) && !tQuestionInfoService.existId(id)) {
            return ResponseResult.error(ErrorCodeEnum.INVALID_RELATED_RECORD_ID);
        }
        if (pageParam != null) {
            PageMethod.startPage(pageParam.getPageNum(), pageParam.getPageSize(), pageParam.getCount());
        }
        TPaperInfo filter = MyModelUtil.copyTo(tPaperInfoDtoFilter, TPaperInfo.class);
        String orderBy = MyOrderParam.buildOrderBy(orderParam, TPaperInfo.class);
        List<TPaperInfo> tPaperInfoList =
                tPaperInfoService.getNotInTPaperInfoListByQuestionId(id, filter, orderBy);
        return ResponseResult.success(MyPageUtil.makeResponseData(tPaperInfoList, TPaperInfoVo.class));
    }

    /**
     * 列出与指定试卷题目存在多对多关系的 [考试试卷] 列表数据。
     *
     * @param id 主表关联字段。
     * @param tPaperInfoDtoFilter [考试试卷] 过滤对象。
     * @param orderParam 排序参数。
     * @param pageParam 分页参数。
     * @return 应答结果对象，返回符合条件的数据列表。
     */
    @SaCheckPermission("tQuestionInfo.view")
    @PostMapping("/listTPaperQuest")
    public ResponseResult<MyPageData<TPaperInfoVo>> listTPaperQuest(
            @MyRequestBody(required = true) Long id,
            @MyRequestBody TPaperInfoDto tPaperInfoDtoFilter,
            @MyRequestBody MyOrderParam orderParam,
            @MyRequestBody MyPageParam pageParam) {
        if (!tQuestionInfoService.existId(id)) {
            return ResponseResult.error(ErrorCodeEnum.INVALID_RELATED_RECORD_ID);
        }
        if (pageParam != null) {
            PageMethod.startPage(pageParam.getPageNum(), pageParam.getPageSize(), pageParam.getCount());
        }
        TPaperInfo filter = MyModelUtil.copyTo(tPaperInfoDtoFilter, TPaperInfo.class);
        String orderBy = MyOrderParam.buildOrderBy(orderParam, TPaperInfo.class);
        List<TPaperInfo> tPaperInfoList =
                tPaperInfoService.getTPaperInfoListByQuestionId(id, filter, orderBy);
        return ResponseResult.success(MyPageUtil.makeResponseData(tPaperInfoList, TPaperInfoVo.class));
    }

    /**
     * 批量添加试卷题目和 [考试试卷] 对象的多对多关联关系数据。
     *
     * @param questionId 主表主键Id。
     * @param tPaperQuestDtoList 关联对象列表。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @OperationLog(type = SysOperationLogType.ADD_M2M)
    @PostMapping("/addTPaperQuest")
    public ResponseResult<Void> addTPaperQuest(
            @MyRequestBody Long questionId,
            @MyRequestBody List<TPaperQuestDto> tPaperQuestDtoList) {
        if (MyCommonUtil.existBlankArgument(questionId, tPaperQuestDtoList)) {
            return ResponseResult.error(ErrorCodeEnum.ARGUMENT_NULL_EXIST);
        }
//        Set<Long> paperIdSet =
//                tPaperQuestDtoList.stream().map(TPaperQuestDto::getPaperId).collect(Collectors.toSet());
//        if (!tQuestionInfoService.existId(questionId)
//                || !tPaperInfoService.existUniqueKeyList("id", paperIdSet)) {
//            return ResponseResult.error(ErrorCodeEnum.INVALID_RELATED_RECORD_ID);
//        }
        List<TPaperQuest> tPaperQuestList =
                MyModelUtil.copyCollectionTo(tPaperQuestDtoList, TPaperQuest.class);
        tQuestionInfoService.addTPaperQuestList(tPaperQuestList, questionId);
        return ResponseResult.success();
    }

    /**
     * 更新指定试卷题目和指定 [考试试卷] 的多对多关联数据。
     *
     * @param tPaperQuestDto 对多对中间表对象。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @OperationLog(type = SysOperationLogType.UPDATE)
    @PostMapping("/updateTPaperQuest")
    public ResponseResult<Void> updateTPaperQuest(
            @MyRequestBody TPaperQuestDto tPaperQuestDto) {
        String errorMessage = MyCommonUtil.getModelValidationError(tPaperQuestDto);
        if (errorMessage != null) {
            return ResponseResult.error(ErrorCodeEnum.DATA_VALIDATED_FAILED, errorMessage);
        }
        TPaperQuest tPaperQuest = MyModelUtil.copyTo(tPaperQuestDto, TPaperQuest.class);
        if (!tQuestionInfoService.updateTPaperQuest(tPaperQuest)) {
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST);
        }
        return ResponseResult.success();
    }

    /**
     * 显示试卷题目和指定 [考试试卷] 的多对多关联详情数据。
     *
     * @param questionId 主表主键Id。
     * @param paperId 从表主键Id。
     * @return 应答结果对象，包括中间表详情。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @GetMapping("/viewTPaperQuest")
    public ResponseResult<TPaperQuestVo> viewTPaperQuest(
            @RequestParam Long questionId, @RequestParam Long paperId) {
        TPaperQuest tPaperQuest = tQuestionInfoService.getTPaperQuest(questionId, paperId);
        if (tPaperQuest == null) {
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST);
        }
        TPaperQuestVo tPaperQuestVo = MyModelUtil.copyTo(tPaperQuest, TPaperQuestVo.class);
        return ResponseResult.success(tPaperQuestVo);
    }

    /**
     * 移除指定试卷题目和指定 [考试试卷] 的多对多关联关系。
     *
     * @param questionId 主表主键Id。
     * @param paperId 从表主键Id。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @OperationLog(type = SysOperationLogType.DELETE_M2M)
    @PostMapping("/deleteTPaperQuest")
    public ResponseResult<Void> deleteTPaperQuest(
            @MyRequestBody Long questionId, @MyRequestBody Long paperId) {
        if (MyCommonUtil.existBlankArgument(questionId, paperId)) {
            return ResponseResult.error(ErrorCodeEnum.ARGUMENT_NULL_EXIST);
        }
        if (!tQuestionInfoService.removeTPaperQuest(questionId, paperId)) {
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST);
        }
        return ResponseResult.success();
    }

    /**
     * 批量移除指定试卷题目和指定 [考试试卷] 的多对多关联关系。
     *
     * @param questionId 主表主键Id。
     * @param paperIdList 从表主键Id列表。
     * @return 应答结果对象。
     */
    @SaCheckPermission("tQuestionInfo.update")
    @OperationLog(type = SysOperationLogType.DELETE_M2M_BATCH)
    @PostMapping("/deleteTPaperQuestList")
    public ResponseResult<Void> deleteTPaperQuestList(
            @MyRequestBody Long questionId, @MyRequestBody List<Long> paperIdList) {
        if (MyCommonUtil.existBlankArgument(questionId, paperIdList)) {
            return ResponseResult.error(ErrorCodeEnum.ARGUMENT_NULL_EXIST);
        }
        tQuestionInfoService.removeTPaperQuestList(questionId, paperIdList);
        return ResponseResult.success();
    }

    /**
     * 以字典形式返回全部试卷题目数据集合。字典的键值为[id, title]。
     * 白名单接口，登录用户均可访问。
     *
     * @param filter 过滤对象。
     * @return 应答结果对象，包含的数据为 List<Map<String, String>>，map中包含两条记录，key的值分别是id和name，value对应具体数据。
     */
    @GetMapping("/listDict")
    public ResponseResult<List<Map<String, Object>>> listDict(@ParameterObject TQuestionInfoDto filter) {
        List<TQuestionInfo> resultList =
                tQuestionInfoService.getListByFilter(MyModelUtil.copyTo(filter, TQuestionInfo.class));
        return ResponseResult.success(
                MyCommonUtil.toDictDataList(resultList, TQuestionInfo::getId, TQuestionInfo::getTitle));
    }

    /**
     * 根据字典Id集合，获取查询后的字典数据。
     *
     * @param dictIds 字典Id集合。
     * @return 应答结果对象，包含字典形式的数据集合。
     */
    @GetMapping("/listDictByIds")
    public ResponseResult<List<Map<String, Object>>> listDictByIds(@RequestParam List<Long> dictIds) {
        List<TQuestionInfo> resultList = tQuestionInfoService.getInList(new HashSet<>(dictIds));
        return ResponseResult.success(
                MyCommonUtil.toDictDataList(resultList, TQuestionInfo::getId, TQuestionInfo::getTitle));
    }

    private ResponseResult<Void> doDelete(Long id) {
        String errorMessage;
        // 验证关联Id的数据合法性
        TQuestionInfo originalTQuestionInfo = tQuestionInfoService.getById(id);
        if (originalTQuestionInfo == null) {
            // NOTE: 修改下面方括号中的话述
            errorMessage = "数据验证失败，当前 [对象] 并不存在，请刷新后重试！";
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST, errorMessage);
        }
        if (!tQuestionInfoService.remove(id)) {
            errorMessage = "数据操作失败，删除的对象不存在，请刷新后重试！";
            return ResponseResult.error(ErrorCodeEnum.DATA_NOT_EXIST, errorMessage);
        }
        return ResponseResult.success();
    }

    @GetMapping("/getQuestionInfoByPaperId")
    public ResponseResult<List<TQuestionInfo>> getQuestionInfoByPaperId(@RequestParam Long paperId) {
        List<TQuestionInfo> tQuestionInfos = tQuestionInfoService.getQuestionInfoByPaperId(paperId);
        return ResponseResult.success(tQuestionInfos);
    }
}
